package idv.arthur.work;

import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.google.gson.Gson;

/**
 * Servlet implementation class DelMRitem
 */ 
public class QueryHost extends HttpServlet {
	private static final long serialVersionUID = 1L;
       
    public QueryHost() {
        super();
        // TODO Auto-generated constructor stub
    }
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    	doPost(request,response);
    }

	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		
		request.setCharacterEncoding("UTF-8");
		
		Widgets wid = new Widgets();
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		String strSQL = "SELECT COUNT(*) FROM hosts";
		ResultTableSet resultTableSet = new ResultTableSet();
		String dataType = request.getParameter("getDataType");
		ArrayList<InfoStruc> infoStrucs = new ArrayList<InfoStruc>();
		int page = Integer.parseInt(request.getParameter("page"));
		int total_pages = 1;
		int count  = 0;
		int limit = Integer.parseInt(request.getParameter("rows"));
		
		try {
			conn = wid.getConn(true);	
			ps = conn.prepareStatement(strSQL);
			rs = ps.executeQuery();
			rs.next();
			count  = rs.getInt(1);
			
			if ( count  != 0 ) {
				
				if (limit  < count) {
					total_pages = (count / limit)+1;	
				}
				
				strSQL = "SELECT t1.hostip, t1.subsysid, t1.bssid, t1.port, t1.hostname,t2.bssname,t3.subsysname FROM hosts t1,bss t2,subsys t3 WHERE t1.bssid = t2.bssid AND t1.subsysid = t3.subsysid ORDER BY hostname ASC";
				ps = conn.prepareStatement(strSQL);
				rs = ps.executeQuery();
				ArrayList<Rows> rows = new ArrayList<Rows>();
				
				while ( rs.next() ) {
					String hostIP = rs.getString("hostIP");
					if ("JSON".equalsIgnoreCase(dataType)) {
						InfoStruc infoStruc = new InfoStruc();
						infoStruc.setId(hostIP);
						infoStruc.setName(rs.getString("hostname"));
						infoStrucs.add(infoStruc);
					} else {
						Rows row = new Rows();
						String[] cell = null;
						row.setId(hostIP);
						if ("mapping".equalsIgnoreCase(dataType)) { 
							cell = new String[6];
							cell[1] = hostIP;
							cell[2] = rs.getString("hostname");
							cell[3] = String.valueOf(rs.getInt("port"));
							cell[4] = rs.getString("bssname")+" ("+rs.getString("bssid")+")";
							cell[5] = rs.getString("subsysname")+" ("+rs.getString("subsysid")+")";
						} else {
							cell = new String[5];
							cell[0] = hostIP;
							cell[1] = rs.getString("hostname");
							cell[2] = String.valueOf(rs.getInt("port"));
							cell[3] = rs.getString("bssname")+" ("+rs.getString("bssid")+")";
							cell[4] = rs.getString("subsysname")+" ("+rs.getString("subsysid")+")";
						}
						row.setCell(cell);
						rows.add(row);
						
					}
				}
				resultTableSet.setRows(rows);
				resultTableSet.setPage(String.valueOf(page));
				resultTableSet.setTotal(total_pages);
				resultTableSet.setRecords(String.valueOf(count));
				
			} else {
				page =0;
				total_pages=0;
				resultTableSet.setPage(String.valueOf(page));
				resultTableSet.setTotal(total_pages);
				resultTableSet.setRecords(String.valueOf(count));
			}
		} catch ( SQLException e) {
			System.out.println(wid.errProcess(e));
		} finally {
			try {
				if (rs != null) {rs.close();}
				if (ps!= null) {ps.close();}
				if (conn!= null) {conn.close();}
			} catch (SQLException e) {
				System.out.println(wid.errProcess(e));
			}
		}
		
		response.setContentType("application/json");
		response.setCharacterEncoding("UTF-8");
		if ("JSON".equalsIgnoreCase(dataType)) {
			response.getWriter().write(new Gson().toJson(infoStrucs));
		} else {
			response.getWriter().write(new Gson().toJson(resultTableSet));	
		}
			
				
	}
}